package leetcode;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

/**
 * 129. 求根到叶子节点数字之和
 * 给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。
 * 例如，从根到叶子节点路径 1->2->3 代表数字 123。
 * 计算从根到叶子节点生成的所有数字之和。
 * 说明: 叶子节点是指没有子节点的节点。
 * 示例 1:
 * 输入: [1,2,3]
 * 1
 * / \
 * 2   3
 * 输出: 25
 * 解释:
 * 从根到叶子节点路径 1->2 代表数字 12.
 * 从根到叶子节点路径 1->3 代表数字 13.
 * 因此，数字总和 = 12 + 13 = 25.
 * 示例 2:
 * 输入: [4,9,0,5,1]
 * 4
 * / \
 * 9   0
 * / \
 * 5   1
 * 输出: 1026
 * 解释:
 * 从根到叶子节点路径 4->9->5 代表数字 495.
 * 从根到叶子节点路径 4->9->1 代表数字 491.
 * 从根到叶子节点路径 4->0 代表数字 40.
 * 因此，数字总和 = 495 + 491 + 40 = 1026.
 */

public class problems_129 {

    public static void main(String[] arge) {
//        TreeNode A1 = new TreeNode(1);
//        TreeNode A2 = new TreeNode(2);
//        TreeNode A3 = new TreeNode(3);
//        TreeNode A4 = new TreeNode(5);
//        TreeNode A5 = new TreeNode(9);
//        A1.left = A2;
//        A1.right = A3;
//        A2.left = A4;
//        A2.right = A5;
        TreeNode A1 = new TreeNode(0);
        TreeNode A2 = new TreeNode(1);
        A1.left = A2;
        System.out.println(new Solution().sumNumbers(A1));
    }

    public static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

    static class Solution {
        List<Stack<Integer>> list = new ArrayList<>();
        public int sumNumbers(TreeNode root) {
            // 遍历取值
            Stack<Integer> stack = new Stack<>();
            if (null != root) dfs(root, stack);
            // 统计
            int sumStream = list.stream().mapToInt((stackObj) -> {
                int sum = 0;
                for (int num : stackObj) {
                    sum = sum * 10 + num;
                }
                return sum;
            }).sum();
            return sumStream;
        }
        //
        public void dfs(TreeNode node, Stack<Integer> stack) {
            stack.push(node.val);
            if (null == node.left && null == node.right) {
                list.add((Stack<Integer>) stack.clone());
                return;
            }
            if (null != node.left) {
                dfs(node.left, stack);
                stack.pop();
            }
            if (null != node.right) {
                dfs(node.right, stack);
                stack.pop();
            }
        }
    }
}
